#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
英语词汇量估算工具启动脚本
"""

import os
import sys
from app import app, db

def main():
    """主函数"""
    print("=" * 50)
    print("英语词汇量估算工具")
    print("=" * 50)
    
    # 检查数据文件
    check_data_files()
    
    # 初始化数据库
    with app.app_context():
        db.create_all()
        print("✓ 数据库初始化完成")
    
    # 启动应用
    print("✓ 系统启动中...")
    print("✓ 访问地址: http://localhost:5000")
    print("✓ 按 Ctrl+C 停止服务")
    print("=" * 50)
    
    app.run(debug=True, host='0.0.0.0', port=5000)

def check_data_files():
    """检查必要的数据文件"""
    required_files = [
        'data/小学英语大纲词汇.txt',
        'data/中考英语词汇表.txt',
        'data/Highschool_edited.txt',
        'data/CET4_edited.txt',
        'data/CET6_edited.txt',
        'data/COCA_frequency_20000.txt'
    ]
    
    missing_files = []
    for file_path in required_files:
        if not os.path.exists(file_path):
            missing_files.append(file_path)
    
    if missing_files:
        print("⚠ 警告: 以下数据文件缺失:")
        for file_path in missing_files:
            print(f"   - {file_path}")
        print("   系统可能无法正常工作，请确保所有数据文件存在。")
    else:
        print("✓ 所有数据文件检查完成")

if __name__ == '__main__':
    main() 